package com.ai.demo.repository;

import org.springframework.stereotype.Component;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
@Component
public class InMemoryChatHistoryRespository implements  ChatHistoryRepository{
    private Map<String,List<String>> chatHistory = new HashMap<String,List<String>>();
    @Override
    public void save(String type, String chatId) {
//        if(!chatHistory.containsKey(type)){
//            chatHistory.put(type,new ArrayList<>());
//        }
//        List <String> chatIds = chatHistory.get(type);
//      下面是简化写法
        List <String> chatIds = chatHistory.computeIfAbsent( type, k -> new ArrayList<>());
        if(!chatIds.contains(chatId)){
            return;
        }
         chatIds.add(chatId);
    }

    @Override
    public List<String> getChatIds(String type) {
        List<String> chatIds = chatHistory.get(type);
        return chatIds==null ? new ArrayList<>():chatIds;
         //return chatHistory.getOrDefault(type,List.of());
    }
}
